package com.cbhb.sys;

import java.util.*;

/**
 * 给定两个数组，编写一个函数来计算它们的交集。
 */
public class InterSection {

    public int[] intersection(int[] nums1, int[] nums2) {
        Set<Integer> set1 = new HashSet<>();
        Set<Integer> set2 = new HashSet<>();
        for(int item:nums1) {
            set1.add(item);
        }
        for(int item:nums2) {
            if(set1.contains(item)) set2.add(item);
        }
        int[] res = new int[set2.size()];
        Iterator<Integer> iterator = set2.iterator();
        for(int i= 0; i< set2.size(); i++) {
            res[i] = iterator.next();
        }
        return res;
    }

    public static void main(String[] args) {
        System.out.println(new InterSection().intersection(new int[]{4, 9, 5}, new int[]{9, 4, 9, 8, 4}));
    }
}
